from quick_sort import quick_sort
from quick_sort_generator import generate_random_numbers


def test_quick_sort():
    # 生成测试数据
    test_data = generate_random_numbers(10)
    print("原始数据:", test_data)
    
    # 执行快速排序
    sorted_data = quick_sort(test_data)
    print("排序后数据:", sorted_data)
    
    # 验证排序结果
    assert sorted_data == sorted(test_data), "排序结果不正确！"
    print("测试通过！排序结果正确。")

if __name__ == "__main__":
    # 运行测试
    test_quick_sort()